import numpy
from   setuptools  import setup
from   torch.utils.cpp_extension import CppExtension, BuildExtension

setup(name='gelu', version="v1.0.0", 
    author="yangxian",
    author_email="yangxianpku@pku.edu.cn",
    ext_modules=[CppExtension(name='gelu', 
                    sources=['gelu.cc'])
                ],
    cmdclass={'build_ext': BuildExtension}
)
